package com.tanhua.server.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.common.pojo.User;
import com.tanhua.common.pojo.UserInfo;
import com.tanhua.common.utils.UserThreadLocal;
import com.tanhua.dubbo.server.pojo.RecommendUser;
import com.tanhua.dubbo.server.vo.PageInfo;
import com.tanhua.server.vo.PageResult;

import com.tanhua.server.vo.RecommendUserQueryParam;
import com.tanhua.server.vo.TodayBest;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;

@Service
public class TodayBestService {
    @Autowired
    private UserService userService;

    @Autowired
    private RecommendUserService recommendUserService;

    @Value("${tanhua.sso.default.user}")
    private Long defaultUserId;

    @Autowired
    private UserInfoService userInfoService;

    public TodayBest queryTodayBest() {
        User user = UserThreadLocal.get();
        // 查询推荐用户(今日佳人)
        TodayBest todayBest = recommendUserService.queryTodayBest(user.getId());
        // 首页必须有显示,如果查到null,这里给出默认的推荐
        if (null == todayBest) {
            todayBest = new TodayBest();
            todayBest.setId(defaultUserId);
            // 固定默认缘分值
            todayBest.setFateValue(80L);
        }

        // 补全个人信息
        UserInfo userInfo = userInfoService.queryUserInfoByUserId(todayBest.getId());
        if (null == userInfo) {
            return null;
        }
        todayBest.setAvatar(userInfo.getLogo());
        todayBest.setNickname(userInfo.getNickName());
        todayBest.setTags(StringUtils.split(userInfo.getTags(), ","));
        todayBest.setGender(userInfo.getSex().getValue() == 1 ? "man" : "woman");
        todayBest.setAge(userInfo.getAge());
        return todayBest;
    }

    /**
     * 查询推荐用户列表
     *
     * @param queryParam 查询条件对象
     * @return PageResult
     */
    public PageResult queryRecommendation(RecommendUserQueryParam queryParam) {
        // 校验token
        User user = UserThreadLocal.get();

        PageResult pageResult = new PageResult();
        pageResult.setPage(queryParam.getPage());
        pageResult.setPageSize(queryParam.getPagesize());

        // 去MongoDB中查询推荐用户列表
        PageInfo<RecommendUser> pageInfo = recommendUserService.queryRecommendUserList(user.getId(), queryParam.getPage(), queryParam.getPagesize());
        List<RecommendUser> records = pageInfo.getRecords();
        if (CollectionUtils.isEmpty(records)) {
            // 没有查到,返回一个空的推荐用户列表
            return pageResult;
        }

        // 填充个人信息
        // 去MySQL中收集推荐用户的id
        HashSet<Long> userIds = new HashSet<>();
        for (RecommendUser record : records) {
            userIds.add(record.getUserId());
        }
        QueryWrapper<UserInfo> queryWrapper = new QueryWrapper<>();
        queryWrapper.in("user_id", userIds);
        if (StringUtils.isNotEmpty(queryParam.getGender())) {
            // 需要性别参数查询
            queryWrapper.eq("sex", StringUtils.equals(queryParam.getGender(), "man") ? 1 : 2);
        }
        if (StringUtils.isNotEmpty(queryParam.getCity())) {
            // 需要城市参数查询
            queryWrapper.like("city", queryParam.getCity());
        }
        if (queryParam.getAge() != null) {
            // 设置年龄参数，条件：小于等于
            queryWrapper.le("age", queryParam.getAge());
        }
        // 去MySQL中查找用户信息
        List<UserInfo> userInfoList = userInfoService.queryUserInfoList(queryWrapper);
        if (CollectionUtils.isEmpty(userInfoList)) {
            // 没有查询到用户的基本信息
            return pageResult;
        }

        // 将用户信息填充到每一个被推荐用户
        ArrayList<TodayBest> todayBests = new ArrayList<>();
        for (UserInfo userInfo : userInfoList) {
            TodayBest todayBest = new TodayBest();

            todayBest.setId(userInfo.getId());
            todayBest.setAvatar(userInfo.getLogo());
            todayBest.setNickname(userInfo.getNickName());
            todayBest.setTags(StringUtils.split(userInfo.getTags(), ","));
            todayBest.setGender(userInfo.getSex().getValue() == 1 ? "man" : "woman");
            todayBest.setAge(userInfo.getAge());

            // 缘分值
            for (RecommendUser record : records) {
                if (record.getUserId().longValue() == userInfo.getUserId().longValue()) {
                    double score = Math.floor(record.getScore());
                    todayBest.setFateValue(Double.valueOf(score).longValue());
                    break;
                }
            }
            todayBests.add(todayBest);
        }

        // 按照缘分值进行倒序排序
        todayBests.sort((o1, o2) -> (int) (o2.getFateValue() - o1.getFateValue()));

        pageResult.setItems(todayBests);
        return pageResult;
    }
}